package com.fuyun.rpc.client.loadbalancing;

import com.fuyun.rpc.common.domain.ServiceMetaInfo;

import java.util.List;
import java.util.Map;
import java.util.concurrent.atomic.AtomicInteger;

/**
 * 轮询
 *
 * @author: zrq1351
 * @date: 2025/2/16
 */
public class RoundRobinLoadBalancer implements LoadBalancer {

	/**
	 * 当前轮询的下标
	 */
	private final AtomicInteger currentIndex = new AtomicInteger(0);

	/**
	 * 选择服务的策略方法 该方法根据当前索引和增刊的列表大小，使用取模算法来选择一个服务 如果列表为空，则返回null；如果列表只有一个服务，则直接返回该服务 对于多个服务的情况，使用取模算法实现轮询选择
	 *
	 * @param requestParams       请求参数，本方法中未使用，但可能在扩展时用于更复杂的逻辑
	 * @param serviceMetaInfoList 服务元信息列表，用于选择服务
	 * @return 根据选择算法选出的服务元信息，如果没有选中则返回null
	 */
	@Override
	public ServiceMetaInfo select(Map<String, Object> requestParams, List<ServiceMetaInfo> serviceMetaInfoList) {
		if (serviceMetaInfoList.isEmpty()) {
			return null;
		}
		// 只有一个服务，无需轮询
		int size = serviceMetaInfoList.size();
		if (size == 1) {
			return serviceMetaInfoList.get(0);
		}
		// 取模算法轮询
		int index = currentIndex.getAndIncrement() % size;
		return serviceMetaInfoList.get(index);
	}
}